import pandas as pd
from tqdm import tqdm
import numpy as np
佣金表=pd.read_excel(r".\客户佣金率.xlsx")

def 佣金信息(客户,品种):
    if 品种=="股票" :
        净佣金率=佣金表[佣金表["客户"]==客户]["股票"].iloc[-1]/1000
    elif 品种=="ETF" :
        净佣金率=佣金表[佣金表["客户"]==客户]["ETF"].iloc[-1]/1000
    elif 品种=="深市转债" :
        净佣金率=佣金表[佣金表["客户"]==客户]["深市转债"].iloc[-1]/1000
    elif 品种=="沪市转债" :
        净佣金率=佣金表[佣金表["客户"]==客户]["沪市转债"].iloc[-1]/1000
    return 净佣金率
交易明细=pd.read_csv(r".\1-3.csv")
for index,row in tqdm(交易明细.iterrows()):
    if pd.isna(row["净佣金率"])==True:
        if row["品种"]=="股票":
            净佣金率=佣金信息(row["客户"],"股票")
            print(净佣金率,type(净佣金率))
            交易明细.iloc[index,"净佣金率"]=净佣金率
        elif row["品种"]=="ETF":
            净佣金率=佣金信息(row["客户"],"ETF")
            交易明细.iloc[index,"净佣金率"]=净佣金率
        elif row["品种"]=="可转债" and row["交易所"]=="深圳证券交易所":
            净佣金率=佣金信息(row["客户"],"深市转债")
            交易明细.iloc[index,"净佣金率"]=净佣金率
        elif row["品种"]=="可转债" and row["交易所"]=="上海证券交易所":
            净佣金率=佣金信息(row["客户"],"沪市转债")
            交易明细.iloc[index,"净佣金率"]=净佣金率

交易明细["净佣金"]=交易明细["净佣金率"]*交易明细["成交金额"]
交易明细.to_csv(r".\运行结果.csv")
